package com.cui.project.model.enums;


import lombok.Getter;

/**
 * 预测模型类型
 */
@Getter
public enum ForecastModelTypeEnum {

    prophet("prophet","Prophet"),
    LSTM("lstm" ,"LSTM"),
    ARIMA("arima" ,"ARIMA"),
    ENSEMBLE("ensemble","集成模型" );
    private final String name;
    private final String description;

    ForecastModelTypeEnum(String name, String description) {
        this.name = name;
        this.description = description;

    }


    public static ForecastModelTypeEnum get(String name){
        for (ForecastModelTypeEnum value : values()) {
            if(value.name.equals(name)){
                return value;
            }
        }
        return null;
    }




}
